package main;

import mulan.data.MultiLabelInstances;
import weka.classifiers.trees.J48;
import classifiers.KillSafeBinaryRelevanceModel;
import evaluation.MulanEvaluator;
import evaluation.PerformanceEvaluation;

public class TestKillSafeModelBuilder {
	
	public static void main(String[] args) throws Exception {
		final String xml_file = "xml files//mulan_clusters.xml";
		final String train_file  = "train_2000_nom.arff";
		final String test_file = "test_1000_nom.arff";
		final String out_file = "res_2000_nom_no_weights_DT.txt";
		KillSafeBinaryRelevanceModel builder = new KillSafeBinaryRelevanceModel(new J48());
		MultiLabelInstances train = new MultiLabelInstances(test_file,xml_file);
		builder.trainModels(train);
		MulanEvaluator me = new MulanEvaluator(test_file,xml_file,builder.getConcurrentBinaryRelevance(train,.01));
		PerformanceEvaluation pe = me.evaluate();
		pe.flushToFile(out_file);
	}

}
